# -*- coding: utf-8 -*-
# Copyright (c) Facebook, Inc. and its affiliates.
from detectron2.config.config import CfgNode

def add_relation_config(cfg):
    """
    Add config for DeepLab.
    """
    cfg.MODEL.RELATION = CfgNode()
    cfg.MODEL.RELATION.THRESH = 0.5
    cfg.MODEL.RELATION.N_OBJECT_PER_IMAGE = 100
    cfg.MODEL.RELATION.N_R = 64
    cfg.MODEL.RELATION.D_F = 128
    cfg.MODEL.RELATION.D_G = 64
    cfg.MODEL.RELATION.D_K = 64
